package com.tricky.demo.rabbit.handler;

import cn.hutool.log.Log;
import lombok.extern.log4j.Log4j;
import lombok.extern.log4j.Log4j2;
import org.apache.ibatis.annotations.Case;
import org.springframework.amqp.core.Message;
import org.springframework.amqp.rabbit.connection.CorrelationData;
import org.springframework.amqp.rabbit.core.RabbitTemplate;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;

import javax.annotation.PostConstruct;

/**
 * @ClassName ComfirmCallback
 * @Description 自定义消息发送确认回调
 * @Author HP
 * @Time 2022/11/24 11:29
 */
@Component
@Log4j2
public class ComfirmCallback implements RabbitTemplate.ConfirmCallback, RabbitTemplate.ReturnCallback {

    @Autowired
    private RabbitTemplate rabbitTemplate;

    @PostConstruct // 用于在依赖关系注入完成之后需要执行的方法上，即程序启动后，执行初始化
    public void init() {
        log.info("***********回调初始化*********");
        rabbitTemplate.setConfirmCallback(this);
        rabbitTemplate.setReturnCallback(this);
    }

    /**
     * Confirmation callback.
     * <p>
     * 确认消息是否成功到达交换机中，不管是否到达交换机，该回调都会执行
     * 生产者 → 交换机
     */
    @Override
    public void confirm(CorrelationData correlationData, boolean ack, String casue) {
        log.info("发布消息的UUID为 {}", correlationData.getId());
        if (ack) {
            log.info("消息发布成功");
        } else {
            log.info("消息发布失败，失败原因为：{}", casue);
        }
    }

    /**
     * Returned message callback.
     * <p>
     * 确认消息是否从交换机成功到达队列中，失败将会执行，成功则不执行
     * 交换机 → 队列
     */
    @Override
    public void returnedMessage(Message message, int i, String s, String s1, String s2) {
        log.info("ReturnsCallback 回调内容：{}", message);
    }
}
